Fixed small bug in #375