Bug fix return value