changes to fix to pass tests