make the solution work as expected