Bug fix

#10, #11