Fixing build error

Sorry made a mistake