Fixed small bug in interface