the functionality works, but tests don't pass