restore old logic