Fixed stupid bug

Added return statement.