minor fix (return instead of assign)