Small code review

#180