slight fix to solve