fix missing functionality