fix varius bugs