Added a small fix regarding #56