Fixed wrong return statements.