small fix for v 0.004