Minor bug fix

Just need to double check on server, hence the pull request for minor fix.