work correctly now