Of course it's ok to book a return flight

However, if you are allowed to use the "return" bit is down to what visa you have at the time of the return.

.... and if u can't, I'd check if there is a refund poicy in operation.