fix: adapt for 0.5.1