fixing broken build - 0.5.4