Fix till #4
Fix till #4