Trying to fix bug

Maybe?