{
        struct octeon_mbox *mbox = oct->mbox[mbox_cmd->q_no];
        u32 count, i, ret = OCTEON_MBOX_STATUS_SUCCESS;
+       long timeout = LIO_MBOX_WRITE_WAIT_TIME;
        unsigned long flags;
 
        spin_lock_irqsave(&mbox->lock, flags);
        count = 0;
 
        while (readq(mbox->mbox_write_reg) != OCTEON_PFVFSIG) {
-               schedule_timeout_uninterruptible(LIO_MBOX_WRITE_WAIT_TIME);
+               schedule_timeout_uninterruptible(timeout);
                if (count++ == LIO_MBOX_WRITE_WAIT_CNT) {
                        ret = OCTEON_MBOX_STATUS_FAILED;
                        break;
                        count = 0;
                        while (readq(mbox->mbox_write_reg) !=
                               OCTEON_PFVFACK) {
-                               schedule_timeout_uninterruptible(10);
+                               schedule_timeout_uninterruptible(timeout);
                                if (count++ == LIO_MBOX_WRITE_WAIT_CNT) {
                                        ret = OCTEON_MBOX_STATUS_FAILED;
                                        break;
 
 #define OCTEON_PFVFSIG                 0x1122334455667788
 #define OCTEON_PFVFERR                 0xDEADDEADDEADDEAD
 
-#define LIO_MBOX_WRITE_WAIT_CNT          1000
-#define LIO_MBOX_WRITE_WAIT_TIME           10
+#define LIO_MBOX_WRITE_WAIT_CNT         1000
+#define LIO_MBOX_WRITE_WAIT_TIME        msecs_to_jiffies(1)
 
 enum octeon_mbox_cmd_status {
        OCTEON_MBOX_STATUS_SUCCESS = 0,